package com.shanzhi.dify.utils;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.shanzhi.dify.entity.ChunkEntity;
import com.shanzhi.dify.entity.DocumentFileEntity;
import com.shanzhi.dify.entity.RetrievalModelRequestEntity;

import java.util.*;

public class DifyDistillUtils {

    /**
     * 从JSON字符串中提取 text.items 内容
     *
     * @param jsonString 完整的JSON响应字符串
     * @return 包含 items 的 JSONArray
     */
    public static JSONArray extractTextItems(String jsonString) {
        JSONObject jsonObject = JSONObject.parseObject(jsonString);
        JSONObject data = jsonObject.getJSONObject("data");
        JSONObject outputs = data.getJSONObject("outputs");
        Object textObj = outputs.get("text");
        if (textObj instanceof JSONObject) {
            JSONObject textJson = (JSONObject) textObj;
            // 处理 JSONObject 情况
            return textJson.getJSONArray("items");
        } else if (textObj instanceof JSONArray) {
            return (JSONArray) textObj;
        } else {
            // 处理其他情况
            return new JSONArray();
        }
    }






}
